package com.lab.project.visitor.verification.utils;

import com.lab.common.utils.DateUtils;
import com.lab.common.utils.security.Md5Utils;
import com.lab.common.utils.sign.Base64;
import lombok.extern.slf4j.Slf4j;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;
import sun.misc.BASE64Encoder;

import java.io.BufferedOutputStream;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;

/**
 * 图片工具类
 *
 * @author wangyanlei
 * @date 2020年7月15日
 */
@Slf4j
public class Base64Save {
    /**
     * 类变量用于生成文件名
     */
    private static int counter = 0;

    /**
     * base64 编码存为图片
     *
     * @param imageStr base64字符串
     * @param filePath 路径
     * @param fileName 名称
     * @return 是否存储成功
     */
    public static Boolean saveImage(String imageStr, String filePath, String fileName) {
        // 对字节数组字符串进行Base64解码并生成图片
        // 图像数据为空
        if (null == imageStr || "".equals(imageStr)) {
            // 返回结果
            return false;
        }
        // 获取图片字节数组
        byte[] a = Base64.decode(imageStr);
        // 字节写成图片
        getFileByBytes(a, filePath, fileName);
        // 返回结果
        return true;
    }

    /**
     * 字节写成图片
     *
     * @param bytes    图片字节流
     * @param filePath 路径
     * @param fileName 文件名
     */
    public static Boolean getFileByBytes(byte[] bytes, String filePath, String fileName) {
        // 缓冲区输出流
        BufferedOutputStream bos = null;
        // 文件输出流
        FileOutputStream fos = null;
        // 文件
        File file;
        try {
            // 设置文件
            File dir = new File(filePath + File.separator + fileName);
            // 判断文件目录是否存在
            if (!dir.getParentFile().exists()) {
                // 创建文件夹
                dir.getParentFile().mkdirs();
            }
            // 判断文件目录是否存在
            if (!dir.exists() && dir.isDirectory()) {
                // 创建文件夹
                dir.mkdirs();
            }
            // 获取文件
            file = new File(dir.getPath());
            // 输出流
            fos = new FileOutputStream(file);
            // 缓冲流
            bos = new BufferedOutputStream(fos);
            // 将字节数组写出
            bos.write(bytes);
            // 返回结果
            return true;
        } catch (Exception e) {
            // 日志
            log.error(e.getMessage());
            // 返回结果
            return false;
        } finally {
            // 输出流不为空
            if (bos != null) {
                try {
                    // 关闭输出流
                    bos.close();
                } catch (IOException e) {
                    // 日志
                    log.error(e.getMessage());
                }
            }
            // 输出流不为空
            if (fos != null) {
                try {
                    // 关闭输出流
                    fos.close();
                } catch (IOException e) {
                    // 日志
                    log.error(e.getMessage());
                }
            }
        }
    }

    /**
     * 编码文件名
     *
     * @param fileName 图片原始名称
     * @return 不重复的图片随机名称+带时间的文件夹 例如： /2020/07/29/5925cb61f35947067bc4598da16d8071.jpg
     */
    public static String extractFilename(String fileName) {
        // 获取随机文件名
        String extension = fileName.substring(fileName.lastIndexOf(".") + 1);
        // 返回文件夹
        return DateUtils.datePath() + "/" + encodingFilename(fileName) + "." + extension;
    }

    /**
     * 根据时间获得MD5随机串
     *
     * @param fileName 图片原始名称
     * @return 文件名
     */
    private static String encodingFilename(String fileName) {
        // 文件名
        fileName = fileName.replace("_", " ");
        // 返回随机穿
        return Md5Utils.hash(fileName + System.nanoTime() + counter++);
    }

    /**
     * 本地图片转换成base64字符串
     *
     * @param imageName 图片名称
     * @return 图片Base64
     * @throws IOException io异常
     */
    public static String ImageToBase64ByLocal(String imageName) throws IOException {
        // 输入流
        InputStream in;
        // 字节数组
        byte[] data = null;
        // 读取图片字节数组
        try {
            //获取图片路径
            Resource resource = new ClassPathResource(imageName);
            // 获取文件
            File file = resource.getFile();
            // 获取文件输入流
            in = new FileInputStream(file.getPath());
            // 文件数据
            data = new byte[in.available()];
            // 读取文件
            in.read(data);
            // 关闭输入流
            in.close();
        } catch (IOException e) {
            // 日志
            log.error(e.getMessage());
        }
        // 对字节数组Base64编码
        BASE64Encoder encoder = new BASE64Encoder();
        // 返回Base64编码过的字节数组字符串
        return encoder.encode(data);
    }

    /**
     * 在线图片转换成流
     *
     * @param imgUrl 图片线上路径
     * @return byte[] 图片流
     */
    public static byte[] imageToByteBufferByOnline(String imgUrl) {
        // 输出流
        ByteArrayOutputStream data = new ByteArrayOutputStream();
        try {
            // 创建URL
            byte[] by = new byte[1024];
            // 获取url
            URL url = new URL(imgUrl);
            // 创建链接
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            // 设置超时时间
            conn.setConnectTimeout(5000);
            // 设置请求方法
            conn.setRequestMethod("GET");
            // 输入流
            // 将内容读取内存中
            int len;
            // 读取文件输入流
            InputStream is = conn.getInputStream();
            // 遍历
            while (-1 != (len = is.read(by))) {
                // 写入文件
                data.write(by, 0, len);
            }
            // 关闭流
            is.close();
        } catch (IOException e) {
            // 日志
            log.error(e.getMessage());
        }
        // 返回结果
        return data.toByteArray();
    }

    /**
     * 在线图片转换成base64字符串
     *
     * @param imgUrl 图片线上路径
     * @return 图片base64字符串
     */
    public static String imageToBase64ByOnline(String imgUrl) {
        // 输出流
        ByteArrayOutputStream data = new ByteArrayOutputStream();
        try {
            // 创建URL
            URL url = new URL(imgUrl);
            // 字节数组
            byte[] by = new byte[1024];
            // 创建链接
            HttpURLConnection conn = (HttpURLConnection) url.openConnection();
            // 请求方法
            conn.setRequestMethod("GET");
            // 超时时间
            conn.setConnectTimeout(5000);
            // 输入流
            InputStream is = conn.getInputStream();
            // 将内容读取内存中
            int len;
            // 遍历
            while ((len = is.read(by)) != -1) {
                // 写入文件
                data.write(by, 0, len);
            }
            // 关闭流
            is.close();
        } catch (IOException e) {
            // 日志
            log.error(e.getMessage());
        }
        // 返回数据
        return org.apache.commons.codec.binary.Base64.encodeBase64String(data.toByteArray());
    }
}
